#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""  
@Project : pythonProject
@File : menu.py
@Author : 万能小魏
@Time : 2024/8/16 14:33  
@脚本说明 : 

"""
import woniu.login as login
import woniu.regist as regist
import woniu.select as select
import woniu.dell as dell
import woniu.update as update

def onemenu():
    print("=" * 10, "蜗牛实验室", "=" * 10)
    print("1. 登录")
    print("2. 注册")
    print("3. 查询账号")
    print("0. 退出")
    return input("请选择功能：\n")

def ermenu():
    print("=" * 10, "蜗牛实验室", "=" * 10)
    print("1. 查询所有")
    print("2. 指定ID查询")
    print("3. 指定ID删除")
    print("4. 指定ID修改")
    print("0. 退出")
    return input("请选择功能：\n")

def ermain():
    while True:
        erchoice = ermenu()
        if erchoice == "1":
            select.selectAll()
        elif erchoice == "2":
            select.selectById()
        elif erchoice == "3":
            dell.delById()
        elif erchoice == "4":
            update.updateById()
        elif erchoice == "0":
            exit()
def onemain():
    while True:
        choice = onemenu()
        if choice == "1":
            login.login()
        if choice == "2":
            regist.register()
        if choice == "3":
            select.selectById()
        if choice == "0":
            print("蜗牛网安实验室 再见!")
            exit()